Python dictionary comprehension